import numpy as np

test = np.array([[1, 2, 3],
                 [2, 3, 4],
                 [5, 4, 3],
                 [8, 7, 2]])
# print(test.shape)

# print(np.argmax(test, 0))  # 输出：array([3, 3, 1]

# print(np.argmax(test, 1))  # 输出：array([2, 2, 0, 0]
# 把数组按下图的矩阵模式排列，axis=0就是求矩阵x轴方向所有y轴最大值下标，
# axis=1是求矩阵y轴方向所有x轴最大值的下标。(下面[表情][表情]不显示，所以用l代替)
test1 = np.array([[[1, 2, 3],
                   [2, 3, 4],
                   [5, 4, 3],
                   [8, 7, 2]], [[1, 2, 3],
                                [2, 3, 4],
                                [5, 4, 3],
                                [8, 7, 2]]])
print(test1.shape)
print(np.argmax(test1, 0))
print((np.argmax(test1, 0).shape))

print(np.argmax(test1, 1))
print((np.argmax(test1, 1).shape))
print(np.argmax(test1, 2))
print((np.argmax(test1, 2).shape))
